#ifndef TRTX_CUDA_UTILS_H_
#define TRTX_CUDA_UTILS_H_

#include <cuda_runtime_api.h>

#ifndef CUDA_CHECK
#define CUDA_CHECK(callstr)                                                \
    {                                                                      \
        cudaError_t error_code = callstr;                                  \
        if (error_code != cudaSuccess) {                                   \
            std::cerr << "CUDA error " << error_code << " at " << __FILE__ \
                      << ":" << __LINE__;                                  \
            assert(0);                                                     \
        }                                                                  \
    }
#endif   // CUDA_CHECK

#endif   // TRTX_CUDA_UTILS_H_
